-
Notifications
You must be signed in to change notification settings - Fork 5.3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Added Differential Diffusion to Auraflow. #9597
base: main
Are you sure you want to change the base?
Conversation
cc @asomoza or @a-r-r-o-w for a review! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thank you for working on this! Just some minor comments, and could you post your results here too?
f" {negative_prompt_attention_mask.shape}." | ||
) | ||
|
||
def encode_prompt( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Could you add a # Copied from
here?
max_sequence_length: int = 256, | ||
output_type: Optional[str] = "pil", | ||
return_dict: bool = True, | ||
map: PipelineImageInput = None, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think docstring of this is missing
def denoising_value_valid(dnv): | ||
return isinstance(dnv, float) and 0 < dnv < 1 | ||
|
||
# sigmas = np.linspace(1.0, 1 / num_inference_steps, num_inference_steps) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
# sigmas = np.linspace(1.0, 1 / num_inference_steps, num_inference_steps) |
# timesteps, num_inference_steps = self.get_timesteps( | ||
# num_inference_steps, | ||
# strength, | ||
# device, | ||
# denoising_start=self.denoising_start if denoising_value_valid(self.denoising_start) else None, | ||
# ) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why has this been removed? If I understand correctly from other implementations, this is required, yes?
# compute the previous noisy sample x_t -> x_t-1 | ||
latents = self.scheduler.step(noise_pred, t, latents, return_dict=False)[0] | ||
|
||
# call the callback, if provided |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Were the callback related parts removed here?
What does this PR do?
Fixes partially #8924
Before submitting
documentation guidelines, and
here are tips on formatting docstrings.
How to test
Who can review?
@a-r-r-o-w @DN6